package com.zzg.algorithm.kmp;

import java.util.Arrays;

/**
 * @Author zhengzg
 * @Date 2022-10-09
 * @Version v1.0
 * KMP算法：字符串匹配问题
 */
public class KmpAlgorithm {
    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";

        int[] next = kemNext(str2);
        System.out.println("next=" + Arrays.toString(next));

        int index = kmpSearch(str1,str2,next);
        System.out.println("index=" + index);
    }

    /**
     * kmp算法
     * @param str1 源字符串
     * @param str2 子串
     * @param next 部分匹配表
     * @return 成功返回索引，否则-1
     */
    private static int kmpSearch(String str1, String str2, int[] next) {
        for (int i = 0,j = 0; i < str1.length(); i++) {
            // 处理str1.charAt(i) != str2.charAt(i) ，调整j的大小
            while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
                j = next[j-1];
            }
            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }
            if (j == str2.length()) {
                // 长度相同证明匹配完成，返回索引位置
                return i - j + 1;
            }
        }
        return -1;
    }

    // 得到字符串的部分匹配表
    private static int[] kemNext(String dest) {
        // 创建一个nexit数组保存部分匹配值
        int[] next = new int[dest.length()];
        // 如果字符串的长度是1，部分匹配值就是0
        next[0] = 0;
        for (int i = 1,j = 0;i < dest.length(); i++) {
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j-1];
            }
            // 当desc.charAt(i) == desc.charAt(j) 满足时，部分匹配值就是+1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
